package backtracking;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

/**
 * @Author: 海琳琦
 * @Date: 2022/2/25 15:16
 * https://leetcode-cn.com/problems/combinations/
 * 组合
 */
public class Combine {
    static List<List<Integer>> result;
    static LinkedList<Integer> list;
    /**
     * 给定两个整数 n 和 k，返回范围 [1, n] 中所有可能的 k 个数的组合。
     * 你可以按任何顺序返回答案。
     * @param n
     * @param k
     * @return
     */
    public static List<List<Integer>> combine(int n, int k) {
        result = new ArrayList<>();
        list = new LinkedList<>();
        traversal(n, 1, k);
        return result;
    }

    /**
     *
     * @param n
     * @param cur  当前处理到的位置，表示集合从哪里开始遍历
     * @param k
     * @return
     */
    public static void traversal(int n, int cur, int k) {
        if (list.size() == k) {
            result.add(new ArrayList<>(list));
            return;
        }
        //剪枝  i最大可以取到 n - (k - list.size()) + 1
        //k - list.size()  表示 还需要选择几个数
        for (int i = cur; i <= n - (k - list.size()) + 1; i++) {
            list.add(i);
            traversal(n, i + 1, k);
            list.removeLast();
        }
    }

    public static void main(String[] args) {
        combine(4, 2);
    }
}
